Image Augumentations

Import Libraries

In [2]:
import imageio
import imgaug as ia
%matplotlib inline
from imgaug import augmenters as iaa
Import sample image for augumentation purpose

In [4]:
image = imageio.imread("/home/jayanthikishore/Downloads/ML_classwork/Week5/image1.png")

print("Original:")
ia.imshow(image)
Original:
Image rotation left and right direction with (360/30) times

In [5]:
newimges =[]
for i in range(12):
    j=i+1
    print(j,j*30)
    rotate = iaa.Affine(rotate=(-30*j, 25))
    rotate1 = iaa.Affine(rotate=(-25,30*j))
    image_aug = rotate(image=image)
    image_aug1 = rotate1(image=image)
    newimges.append(image_aug)
    newimges.append(image_aug1)
    ia.imshow(image_aug1)
1 30
2 60
3 90
4 120
5 150
6 180
7 210
8 240
9 270
10 300
11 330
12 360
In [6]:
print(len(newimges))
24
Gaussian Noise

In [7]:
seq = iaa.Sequential([
    iaa.Affine(rotate=(-25, 25)),
    iaa.AdditiveGaussianNoise(scale=(30, 90)),
    iaa.Crop(percent=(0, 0.4))
], random_order=True)

images_aug = [seq(image=image) for _ in range(20)]

print("Augmented:")
ia.imshow(ia.draw_grid(images_aug, cols=6, rows=2))
Augmented:
Checking shape of the Image

In [8]:
import numpy as np
np.shape(newimges),np.shape(images_aug)
Out[8]:
((24, 193, 290, 4), (20, 193, 290, 4))
Concatinate all the augumented images

In [9]:
concimges = np.concatenate((newimges, images_aug), axis=0)
np.shape(concimges)
Out[9]:
(44, 193, 290, 4)
Save all images into a folder

In [10]:
import cv2 
import os 
trial_dir = "/Users/preethamvignesh/Downloads/ML_classwork/"
j=0
for i in range(len(concimges)):
    pimge = concimges[i]
    j =i +1
    #pfname ="trial_"+str(i) + ".png"  #png is 5*jpg size
    pfname ="augumented_"+str(j) + ".jpg"
    #print(pfname)
    fname =trial_dir + pfname
    print(fname)
    #cv2.imwrite(fname,pimge)
    cv2.imwrite(fname,cv2.cvtColor(pimge, cv2.COLOR_RGB2BGR))
    
    i+=1
    ia.imshow(pimge)
/Users/preethamvignesh/Downloads/ML_classwork/augumented_1.jpg
/Users/preethamvignesh/Downloads/ML_classwork/augumented_2.jpg
/Users/preethamvignesh/Downloads/ML_classwork/augumented_3.jpg
/Users/preethamvignesh/Downloads/ML_classwork/augumented_4.jpg
/Users/preethamvignesh/Downloads/ML_classwork/augumented_5.jpg
/Users/preethamvignesh/Downloads/ML_classwork/augumented_6.jpg
/Users/preethamvignesh/Downloads/ML_classwork/augumented_7.jpg
/Users/preethamvignesh/Downloads/ML_classwork/augumented_8.jpg
/Users/preethamvignesh/Downloads/ML_classwork/augumented_9.jpg
/Users/preethamvignesh/Downloads/ML_classwork/augumented_10.jpg
/Users/preethamvignesh/Downloads/ML_classwork/augumented_11.jpg
/Users/preethamvignesh/Downloads/ML_classwork/augumented_12.jpg
/Users/preethamvignesh/Downloads/ML_classwork/augumented_13.jpg
/Users/preethamvignesh/Downloads/ML_classwork/augumented_14.jpg
/Users/preethamvignesh/Downloads/ML_classwork/augumented_15.jpg
/Users/preethamvignesh/Downloads/ML_classwork/augumented_16.jpg
/Users/preethamvignesh/Downloads/ML_classwork/augumented_17.jpg
/Users/preethamvignesh/Downloads/ML_classwork/augumented_18.jpg
/Users/preethamvignesh/Downloads/ML_classwork/augumented_19.jpg
/Users/preethamvignesh/Downloads/ML_classwork/augumented_20.jpg
/Users/preethamvignesh/Downloads/ML_classwork/augumented_21.jpg
/Users/preethamvignesh/Downloads/ML_classwork/augumented_22.jpg
/Users/preethamvignesh/Downloads/ML_classwork/augumented_23.jpg
/Users/preethamvignesh/Downloads/ML_classwork/augumented_24.jpg
/Users/preethamvignesh/Downloads/ML_classwork/augumented_25.jpg
/Users/preethamvignesh/Downloads/ML_classwork/augumented_26.jpg
/Users/preethamvignesh/Downloads/ML_classwork/augumented_27.jpg
/Users/preethamvignesh/Downloads/ML_classwork/augumented_28.jpg
/Users/preethamvignesh/Downloads/ML_classwork/augumented_29.jpg
/Users/preethamvignesh/Downloads/ML_classwork/augumented_30.jpg
/Users/preethamvignesh/Downloads/ML_classwork/augumented_31.jpg
/Users/preethamvignesh/Downloads/ML_classwork/augumented_32.jpg
/Users/preethamvignesh/Downloads/ML_classwork/augumented_33.jpg
/Users/preethamvignesh/Downloads/ML_classwork/augumented_34.jpg
/Users/preethamvignesh/Downloads/ML_classwork/augumented_35.jpg
/Users/preethamvignesh/Downloads/ML_classwork/augumented_36.jpg
/Users/preethamvignesh/Downloads/ML_classwork/augumented_37.jpg
/Users/preethamvignesh/Downloads/ML_classwork/augumented_38.jpg
/Users/preethamvignesh/Downloads/ML_classwork/augumented_39.jpg
/Users/preethamvignesh/Downloads/ML_classwork/augumented_40.jpg
/Users/preethamvignesh/Downloads/ML_classwork/augumented_41.jpg
/Users/preethamvignesh/Downloads/ML_classwork/augumented_42.jpg
/Users/preethamvignesh/Downloads/ML_classwork/augumented_43.jpg
/Users/preethamvignesh/Downloads/ML_classwork/augumented_44.jpg
In [ ]: